import matplotlib.pyplot as plt
import numpy as np

# 绘制柱状图

x = [2021, 2022, 2023, 2024, 2025]
x_labels = ['2021 年', '2022 年', '2023 年', '2024 年', '2025 年']
y = [1000, 3000, 4000, 5000, 7000]
plt.bar(x, y, width=0.9)
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
plt.xticks(x, x_labels)
plt.xlabel('年份')
plt.ylabel('销量')
plt.title('根据年份销量对比图')
plt.show()
plt.rcParams['axes.unicode_minus'] = False
y = [100, 201, -250, 150]
x = [1, 2, 3, 4]
plt.subplot(1, 2, 1)
plt.bar(x, y, color='blue')
plt.axhline(0, color='blue', linewidth=3)  # 在x为0的地方画一条水平线

plt.subplot(1, 2, 2)
h_bar = plt.barh(x, y, color='red')
plt.axvline(0, color='red', linewidth=3)  # 在x为0的地方画一条垂直线
for bar, height in zip(h_bar, y):
    if height < 0:
        bar.set(edgecolor='darkred', color='lightgreen', linewidth=3)
plt.show()
